#!/usr/bin/env bash

ROOT=/home/ping501b/data/YLIMED/mp4_short_encode
TrainSplit=$ROOT/YLIMED_short_mp4_EvAll_split_train.txt
TestSplit=$ROOT/YLIMED_short_mp4_EvAll_split_test.txt
PosValSplit=$ROOT/YLIMED_short_mp4_Ev101_110_split_val.txt
DEVTrainSplit=$ROOT/YLIMED_short_mp4_EvAll_split_train_dev.txt
DEVTestSplit=$ROOT/YLIMED_short_mp4_EvAll_split_test_dev.txt
TOYSplit=$ROOT/YLIMED_short_mp4_EvAll_split_test_toy.txt

basepath=$(cd `dirname $0`; cd ..; pwd)
NAME=test_environment
LOG=$basepath/out/$NAME/log
if [ ! -d $LOG ];then
    mkdir -p $LOG
fi
MODEL=$basepath/out/$NAME/weights
CUDA_VISIBLE_DEVICES=2,3 \
python $basepath/train.py --arch resnet152 --data-name ylimed \
    --data-root  $ROOT \
    --train-list $DEVTrainSplit \
    --test-list  $DEVTestSplit \
    --representation iframe \
    --model-prefix $MODEL/ylimed \
    --lr        0.0003 \
    --lstm_lr   0.003 \
    --batch-size 32 \
    --lr-steps 40 80 \
    --epochs 90
